package 剑指Offer._6170;

import java.util.Arrays;

/**
 * @author yzh
 * @version 1.0
 * @date 2021/12/14 10:05
 * 构建乘积数组
 * 给定一个数组 A[0,1,…,n-1]，请构建一个数组 B[0,1,…,n-1]，其中 B[i] 的值是数组 A 中除了下标 i 以外的元素的积,
 * 即 B[i]=A[0]×A[1]×…×A[i-1]×A[i+1]×…×A[n-1]。不能使用除法。
 * 算法：前缀和
 * 分别从左右两边求两次前缀和
 */
public class _66 {
    public static void main(String[] args) {
        System.out.println(Arrays.toString(new _66().constructArr(new int[]{1, 2, 3, 4, 5})));
    }

    public int[] constructArr(int[] a) {
        int len = a.length;
        // pre[i - 1] 表示第 i 个元素前所有元素的乘积
        int[] pre = new int[len + 1];
        pre[0] = 1;
        for (int i = 1; i <= len; i++) pre[i] = pre[i - 1] * a[i - 1];
        // tail[i] 表示第 i 个元素后所有元素乘积
        int[] tail = new int[len + 1];
        tail[len] = 1;
        for (int i = len - 1; i >= 0; i--) tail[i] = tail[i + 1] * a[i];
        int[] ans = new int[len];
        for (int i = 1; i <= len; i++) ans[i - 1] = pre[i - 1] * tail[i];
        return ans;
    }

}
